package com.cheetah.start.module.gen.domain;

import cn.hutool.core.collection.ListUtil;
import cn.hutool.core.util.ArrayUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.cheetah.start.common.base.BaseEntity;
import com.cheetah.start.common.constant.GenerateConstant;
import lombok.Data;
import lombok.EqualsAndHashCode;

import java.util.ArrayList;
import java.util.List;

/**
 * 创表
 * Describe: 业务表实体
 * Author: 就 眠 仪 式
 * CreateTime: 2019/10/23
 *
 * @author Administrator
 * @date 2023/05/22
 */
@Data
@EqualsAndHashCode(callSuper = true)
@TableName("gen_table")
public class GenTable extends BaseEntity {

    /** 编号 */
    @TableId("table_id")
    private String tableId;

    /** 表名称 */
    private String tableName;

    /** 表描述 */
    private String tableComment;

    /** 关联父表的表名 */
    private String subTableName;

    /** 本表关联父表的外键名 */
    private String subTableFkName;

    /** 实体类名称(首字母大写) */
    private String className;

    /** 使用的模板（crud单表操作 tree树表操作 sub主子表操作） */
    private String tplCategory;

    /** 生成包路径 */
    private String packageName;

    /** 生成模块名 */
    private String moduleName;

    /** 生成业务名 */
    private String businessName;

    /** 生成功能名 */
    private String functionName;

    /** 生成作者 */
    private String functionAuthor;

    /** 生成代码方式（0zip压缩包 1自定义路径） */
    private String genType;

    /** 生成路径（不填默认项目路径） */
    private String genPath;

    /** 主键信息 */
    @TableField(exist = false)
    private GenTableColumn pkColumn;

    /** 子表信息 */
    @TableField(exist = false)
    private GenTable subTable;

    /** 表列信息 */
    @TableField(exist = false)
    private List<GenTableColumn> columns;

    /** 其它生成选项 */
    private String options;

    /** 树编码字段 */
    private String treeCode;

    /** 树父编码字段 */
    private String treeParentCode;

    /** 树名称字段 */
    private String treeName;

    /** 上级菜单ID字段 */
    private String parentMenuId;

    /** 上级菜单名称字段 */
    private String parentMenuName;

    public static boolean isSub(String tplCategory) {
        return tplCategory != null && StrUtil.equals(GenerateConstant.TPL_SUB, tplCategory);
    }

    public static boolean isTree(String tplCategory) {
        return tplCategory != null && StrUtil.equals(GenerateConstant.TPL_TREE, tplCategory);
    }

    public static boolean isCrud(String tplCategory) {
        return tplCategory != null && StrUtil.equals(GenerateConstant.TPL_CRUD, tplCategory);
    }

    public static boolean isSuperColumn(String tplCategory, String javaField) {
        if (isTree(tplCategory)) {
            String[] treeEntity = GenerateConstant.TREE_ENTITY;
            String[] baseEntity = GenerateConstant.BASE_ENTITY;
            ArrayList<String> trees = ListUtil.toList(treeEntity);
            ArrayList<String> bases = ListUtil.toList(baseEntity);
            trees.addAll(bases);
            String[] strings = trees.toArray(new String[0]);
            return StrUtil.equalsAnyIgnoreCase(javaField, strings);
        }
        return StrUtil.equalsAnyIgnoreCase(javaField, GenerateConstant.BASE_ENTITY);
    }


    public boolean isSub() {
        return isSub(this.tplCategory);
    }

    public boolean isTree() {
        return isTree(this.tplCategory);
    }

    public boolean isCrud() {
        return isCrud(this.tplCategory);
    }

    public boolean isSuperColumn(String javaField) {
        return isSuperColumn(this.tplCategory, javaField);
    }
}
